SPRIG: Semantic-ID-enhanced Paths for Knowledge Graph-based Generative Recommendation
SPRIG:语义ID增强路径的知识图谱生成式推荐
🧭 生成式推荐 · Semantic ID + KG path ★ 深度解读
Johannes Kepler University Linz · Albatross AI · Criteo AI Lab · Linz Institute of Technology
首次在KG路径推理生成式推荐中将物品表示为层级语义ID,统一共享词表,以更少参数实现更强推荐
📌 工作参考价值(4 点)
- 商品/广告ID语义化:用内容embedding + residual K-means生成层级Semantic ID(如3层×256码),替代原子item token,词表从|I|降至固定768,大catalog下可省~24x参数,相似商品共享前缀利于长尾与冷启动。
- 两阶段路径语言建模:先在无协同信号的通用KG路径上预训练,再在用户-物品路径上微调;推理用diverse beam search(60 beams/5组/diversity penalty 0.4)生成路径并按路径log-likelihood排序候选商品。
- 量化器选择:residual K-means优于RVQ/RQ-VAE但差距仅~1%,可优先选简单方案;模型对底层item特征质量敏感,需先保证内容表征质量。
- 若业务有商品知识图谱(类目/品牌/属性关系),值得将KG路径生成与SID结合,而非单独用SID或单独用KG;两者互补,在稀疏数据集上提升明显。